import sys, os
sys.path.append(os.pardir) # 为了导入父目录的文件而设定
import numpy as np
from dataset.mnist import load_mnist
from PIL import Image

def img_show(img):
    pil_img = Image.fromarray(np.uint8(img))
    pil_img.show()

(x_train, t_train),(x_test,t_test) = load_mnist(flatten=True, normalize=False)

print(x_train.shape)
print(t_train.shape)

img = x_train[0]
label = t_train[0]
print(label)
print(img.shape)
img = img.reshape(28,28) # 还原图像原来的尺寸

img_show(img)